from backend.experiment.args import model_type

if 'pangu' in model_type:
    from backend.experiment.pangu import model, context_limit, model_name, \
        vocab_name
elif 'us' in model_type:
    from backend.experiment.us import model, context_limit, model_name, \
        vocab_name
else:
    raise ValueError(f'model_type: {model_type}')
